
import streamlit as st
import data.data as dd
import time
import base64


#设置页面的标签也的名字和icon
st.set_page_config(
    page_title='智能小助手登录页面',
    page_icon='🎶'
)
#页面设计会发光的字体
st.write("""
<style>
.neon-glow {
    font-size: 50px;
    color: #fff;
    text-shadow: 0 0 10px #FFBBFF, 0 0 20px #FFBBFF, 0 0 30px #FFBBFF, 0 0 40px #FFBBFF, 0 0 70px #FFBBFF, 0 0 80px #FFBBFF, 0 0 100px #FFBBFF, 0 0 150px #FFBBFF;
    animation: glow 1s ease-in-out infinite alternate-reverse;
}

@keyframes glow {
    to {
        text-shadow: 0 0 5px #FFBBFF, 0 0 10px #FFBBFF, 0 0 15px #FFBBFF, 0 0 20px #FFBBFF, 0 0 35px #FFBBFF, 0 0 40px #FFBBFF, 0 0 50px #FFBBFF, 0 0 75px #FFBBFF;
    }
}
</style>
""", unsafe_allow_html=True)
st.markdown('<p class="neon-glow">💕鬼马精灵翻墙记💕</p >', unsafe_allow_html=True)
username = st.text_input('请输入用户名')
password = st.text_input('请输入密码',type='password')
col1,col2 = st.columns([2,1])
with col1:
    login_flag = st.button("登录")
def main_bg(main_bg):
    main_bg_ext = "jpg"
    st.markdown(
        f"""
         <style>
         .stApp {{
             background: url(data:image/{main_bg_ext};base64,{base64.b64encode(open(main_bg, "rb").read()).decode()});
             background-size: cover
         }}
         </style>
         """,
        unsafe_allow_html=True)
main_bg('login.jpg')

def login(username, password):
    if username and password:
        result = dd.query_user_by_username(username)
        if result is None:
            st.error('请检查用户名是否书写正确!')
        else:
            if result["password"] == password:
                st.success("登录成功!")
                time.sleep(2)
                st.session_state.user_id = result["user_id"]
                st.session_state.username = result["username"]
                st.switch_page("pages/Intelligent-selection-interface.py")
            else:
                st.error('密码错误,请重新输入!')
    else:
        st.error('信息填写不全!')
#先检验数据是否为空，根据用户名和密码校验数据是否存在
#校验的时候只需要根据用户名查询即可，如果用户名存，再根据获取回来的密码和传入的密码做比对 登录成功之后显示登录成功，延迟两秒到首页(AIz助手聊天界面)
if login_flag:
    login(username, password)



#注册按钮

register_flag = st.button("没有账号？点击注册")
#注册按钮点击之后应该跳转到注册界面
if register_flag:
    #streamlit中有一个函数为switch_page（”py页面文件的路径“）py文件必须位于项目的pages目录下
    st.switch_page("pages/register.py")
with col2:
    uppassword_flag = st.button("忘记密码？点击修改")
if uppassword_flag:
    #streamlit中有一个函数为switch_page（”py页面文件的路径“）py文件必须位于项目的pages目录下
    st.switch_page("pages/uppassword.py")